tensor([[[0.3400, 0.3400],
         [0.3400, 0.3400]],

        [[0.4100, 0.4100],
         [0.4100, 0.4100]],

        [[0.2600, 0.2600],
         [0.2600, 0.2600]]], grad_fn=<RoundBackward1>)